package yunjiao.javatutorials.guava.concurrent.spring;

import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * ResultSet 转 Map/List<Map> 工具类（纯 JDBC 实现，无第三方依赖）
 *
 * @author yangyunjiao
 */
public class ResultSetToMapUtil {
    /**
     * 单行结果：ResultSet -> Map<String, Object>
     * （若结果集有多行，仅返回第一行；无数据返回 null）
     * @param rs 数据库查询结果集
     * @return 单行数据映射（key=列名/别名，value=字段值）
     * @throws SQLException 数据库操作异常
     */
    public static Map<String, Object> resultSetToMap(ResultSet rs) throws SQLException {
        if (rs == null) return null;

        ResultSetMetaData metaData = rs.getMetaData();
        int columnCount = metaData.getColumnCount(); // 总列数

        // 只取第一行数据
        if (rs.next()) {
            Map<String, Object> rowMap = new HashMap<>(columnCount);
            for (int i = 1; i <= columnCount; i++) {
                // 键：优先取 SQL 别名（getColumnLabel），无别名则取列名（getColumnName）
                String columnKey = metaData.getColumnLabel(i);
                // 值：自动适配数据库类型（如 VARCHAR→String、INT→Integer、DATE→Date）
                Object columnValue = rs.getObject(i);
                rowMap.put(columnKey, columnValue);
            }
            return rowMap;
        }
        return null; // 结果集为空
    }

    /**
     * 多行结果：ResultSet -> List<Map<String, Object>>
     * （无数据返回空列表，避免 NPE）
     * @param rs 数据库查询结果集
     * @return 多行数据映射列表
     * @throws SQLException 数据库操作异常
     */
    public static List<Map<String, Object>> resultSetToListMap(ResultSet rs) throws SQLException {
        List<Map<String, Object>> dataList = new ArrayList<>();
        if (rs == null) return dataList;

        ResultSetMetaData metaData = rs.getMetaData();
        int columnCount = metaData.getColumnCount();

        // 遍历所有行
        while (rs.next()) {
            Map<String, Object> rowMap = new HashMap<>(columnCount);
            for (int i = 1; i <= columnCount; i++) {
                String columnKey = metaData.getColumnLabel(i);
                Object columnValue = rs.getObject(i);
                rowMap.put(columnKey, columnValue);
            }
            dataList.add(rowMap);
        }
        return dataList;
    }
}
